import torch
from tensorboardX import SummaryWriter

from model import C3D

model = C3D(num_classes=2)
dummy_input = torch.rand(1, 3, 16, 112, 112)
with SummaryWriter(comment='C3D') as w:
    w.add_graph(model, (dummy_input,))
